package com.apples.mylibrary;

import android.content.ContentUris;
import android.content.Context;
import android.database.Cursor;
import android.net.Uri;
import android.os.Build;
import android.os.Environment;
import android.provider.DocumentsContract;
import android.provider.MediaStore;

import androidx.annotation.NonNull;
import androidx.documentfile.provider.DocumentFile;


/**
 * ******************************************************************************
 * <p><b>Project HexViewer</b><br/>
 * File data representation (used by MainActivity)
 * </p>
 *
 * @author Keidan
 * <p>
 * License: GPLv3
 * </p>
 * ******************************************************************************
 */
public class FileData {
    protected static final String SEQUENTIAL_SEP = "^";
    private final String mName;
    private final Uri mUri;
    private boolean mOpenFromAppIntent;
    private long mStartOffset;
    private long mEndOffset;
    private long mSize;
    private long mRealSize;
    private boolean mIsNotFound;
    private final boolean mIsAccessError;
    private int mShiftOffset;

    private String realPath;

    public String getRealPath() {
        return realPath;
    }

    public void setRealPath(String realPath) {
        this.realPath = realPath;
    }

    public FileData(final Context ctx, final Uri uri, boolean openFromAppIntent) {
        this(ctx, uri, openFromAppIntent, 0L, 0L);
    }

    public FileData(final Context ctx, final Uri uri, boolean openFromAppIntent, long startOffset, long endOffset) {
        mShiftOffset = 0;
        mName = FileHelper.getFileName(ctx, uri);
        mUri = FileHelper.adjustUri(ctx, uri);
        mStartOffset = startOffset;
        mEndOffset = endOffset;
        mOpenFromAppIntent = openFromAppIntent;
        mRealSize = FileHelper.getFileSize(ctx, ctx.getContentResolver(), mUri);
        if (isSequential())
            mSize = Math.abs(mEndOffset - mStartOffset);
        else
            mSize = mRealSize;

        /* We assume that if the file sizes are not <= 0, the file exists. */
        if (Build.VERSION.SDK_INT <= Build.VERSION_CODES.Q && (mSize <= 0 || mRealSize <= 0)) {
            DocumentFile sourceFile = DocumentFile.fromSingleUri(ctx, mUri);
            mIsNotFound = (sourceFile == null || !sourceFile.exists());
        }
        if (mIsNotFound) {
            mIsAccessError = false;
            mRealSize = 0;
            mSize = 0;
        } else {
            if (mRealSize == -1) {
                mIsAccessError = false;
                mIsNotFound = true;
                mRealSize = 0;
                mSize = 0;
            } else if (mRealSize == -2) {
                mIsAccessError = true;
                mIsNotFound = false;
                mRealSize = 0;
                mSize = 0;
            } else {
                mIsAccessError = false;
                mIsNotFound = false;
            }
        }
        Uri docUri = DocumentsContract.buildDocumentUriUsingTree(uri,
                DocumentsContract.getTreeDocumentId(uri));
        realPath = getPath(ctx, docUri);
    }

    /**
     * Sets the offset used to shift the text to the end of the line.
     *
     * @param shiftOffset The new value.
     */
    public void setShiftOffset(int shiftOffset) {
        mShiftOffset = shiftOffset;
    }

    /**
     * Returns the offset used to shift the text to the end of the line.
     *
     * @return int
     */
    public int getShiftOffset() {
        return mShiftOffset;
    }

    /**
     * Returns true if a file access error was raised.
     *
     * @return boolean
     */
    public boolean isAccessError() {
        return mIsAccessError;
    }

    /**
     * Returns true if the file was not found.
     *
     * @return boolean
     */
    public boolean isNotFound() {
        return mIsNotFound;
    }

    /**
     * Tests if the file is opened from the application's intent.
     *
     * @return boolean
     */
    public boolean isOpenFromAppIntent() {
        return mOpenFromAppIntent;
    }

    /**
     * Clears the open from app intent flag.
     */
    public void clearOpenFromAppIntent() {
        mOpenFromAppIntent = false;
    }

    /**
     * Tests if the name is empty.
     *
     * @param fd FileData
     * @return boolean
     */
    public static boolean isEmpty(FileData fd) {
        return fd == null || fd.mName == null || fd.mName.isEmpty();
    }

    /**
     * Returns the file name.
     *
     * @return String
     */
    public String getName() {
        return mName;
    }

    @NonNull
    public String toString() {
        String ret = mStartOffset + SEQUENTIAL_SEP + mEndOffset + SEQUENTIAL_SEP;
        ret += mUri.toString();
        return ret;
    }

    /**
     * Test if it's a sequential file or not.
     *
     * @return boolean
     */
    public boolean isSequential() {
        return mStartOffset != 0L || mEndOffset != 0L;
    }

    /**
     * Returns the file uri.
     *
     * @return Uri
     */
    public Uri getUri() {
        return mUri;
    }

    /**
     * Returns the start offset.
     *
     * @return long
     */
    public long getStartOffset() {
        return mStartOffset;
    }

    /**
     * Returns the end offset.
     *
     * @return long
     */
    public long getEndOffset() {
        return mEndOffset;
    }

    /**
     * Sets the start offset.
     *
     * @param startOffset Start offset
     * @param endOffset   End offset
     * @param refreshSize Refresh the size?
     */
    public void setOffsets(long startOffset, long endOffset, boolean refreshSize) {
        mStartOffset = startOffset;
        mEndOffset = endOffset;
        if (refreshSize)
            mSize = Math.abs(mEndOffset - mStartOffset);
    }

    /**
     * Returns the file size.
     *
     * @return long
     */
    public long getSize() {
        return mSize;
    }

    /**
     * Returns the real file size.
     *
     * @return long
     */
    public long getRealSize() {
        return mRealSize;
    }


    public static String getPath(final Context context, final Uri uri) {

        final boolean isKitKat = Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT;

        // DocumentProvider
        if (isKitKat && DocumentsContract.isDocumentUri(context, uri)) {
            // ExternalStorageProvider
            if (isExternalStorageDocument(uri)) {
                final String docId = DocumentsContract.getDocumentId(uri);
                final String[] split = docId.split(":");
                final String type = split[0];

                if ("primary".equalsIgnoreCase(type)) {
                    return Environment.getExternalStorageDirectory() + "/" + split[1];
                }

                // TODO handle non-primary volumes
            }
            // DownloadsProvider
            else if (isDownloadsDocument(uri)) {

                final String id = DocumentsContract.getDocumentId(uri);
                final Uri contentUri = ContentUris.withAppendedId(
                        Uri.parse("content://downloads/public_downloads"), Long.valueOf(id));

                return getDataColumn(context, contentUri, null, null);
            }
            // MediaProvider
            else if (isMediaDocument(uri)) {
                final String docId = DocumentsContract.getDocumentId(uri);
                final String[] split = docId.split(":");
                final String type = split[0];

                Uri contentUri = null;
                if ("image".equals(type)) {
                    contentUri = MediaStore.Images.Media.EXTERNAL_CONTENT_URI;
                } else if ("video".equals(type)) {
                    contentUri = MediaStore.Video.Media.EXTERNAL_CONTENT_URI;
                } else if ("audio".equals(type)) {
                    contentUri = MediaStore.Audio.Media.EXTERNAL_CONTENT_URI;
                }

                final String selection = "_id=?";
                final String[] selectionArgs = new String[]{
                        split[1]
                };

                return getDataColumn(context, contentUri, selection, selectionArgs);
            }
        }
        // MediaStore (and general)
        else if ("content".equalsIgnoreCase(uri.getScheme())) {

            // Return the remote address
            if (isGooglePhotosUri(uri))
                return uri.getLastPathSegment();

            return getDataColumn(context, uri, null, null);
        }
        // File
        else if ("file".equalsIgnoreCase(uri.getScheme())) {
            return uri.getPath();
        }

        return null;
    }

    public static String getDataColumn(Context context, Uri uri, String selection,
                                       String[] selectionArgs) {

        Cursor cursor = null;
        final String column = "_data";
        final String[] projection = {
                column
        };

        try {
            cursor = context.getContentResolver().query(uri, projection, selection, selectionArgs,
                    null);
            if (cursor != null && cursor.moveToFirst()) {
                final int index = cursor.getColumnIndexOrThrow(column);
                return cursor.getString(index);
            }
        } finally {
            if (cursor != null)
                cursor.close();
        }
        return null;
    }


    /**
     * @param uri The Uri to check.
     * @return Whether the Uri authority is ExternalStorageProvider.
     */
    public static boolean isExternalStorageDocument(Uri uri) {
        return "com.android.externalstorage.documents".equals(uri.getAuthority());
    }

    /**
     * @param uri The Uri to check.
     * @return Whether the Uri authority is DownloadsProvider.
     */
    public static boolean isDownloadsDocument(Uri uri) {
        return "com.android.providers.downloads.documents".equals(uri.getAuthority());
    }

    /**
     * @param uri The Uri to check.
     * @return Whether the Uri authority is MediaProvider.
     */
    public static boolean isMediaDocument(Uri uri) {
        return "com.android.providers.media.documents".equals(uri.getAuthority());
    }

    /**
     * @param uri The Uri to check.
     * @return Whether the Uri authority is Google Photos.
     */
    public static boolean isGooglePhotosUri(Uri uri) {
        return "com.google.android.apps.photos.content".equals(uri.getAuthority());
    }
}
